CODE 10. Best Time to Buy and Sell Stock III

版权声明:本文为博主原创文章,转载请注明出处,谢谢!

版权声明:本文为博主原创文章,转载请注明出处:http://blog.jerkybible.com/2013/09/15/2013-09-15-CODE 10 Best Time to Buy and Sell Stock III/

访问原文「CODE 10. Best Time to Buy and Sell Stock III

Say you have an array for which the ith element
is the price of a given stock on day i.
Design an algorithm to find the maximum profit. You may complete at most two transactions.
Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
public int maxProfit(int[] prices) {
// Start typing your Java solution below
// DO NOT write main() function
if (null == prices || prices.length <= 1) {
return 0;
}
int[] diff = new int[prices.length];
for (int i = 1; i < prices.length; i++) {
diff[i] = prices[i] - prices[i - 1];
}
int maxFinal = 0;
for (int i = 1; i < diff.length; i++) {
if (diff[i] <= 0) {
continue;
}
int max1 = 0;
int max2 = 0;
int tmp = 0;
for (int j = 0; j < i; j++) {
tmp += diff[j];
if (tmp > max1) {
max1 = tmp;
}
if (tmp < 0) {
tmp = 0;
}
}
tmp = 0;
for (int j = i; j < diff.length; j++) {
tmp += diff[j];
if (tmp > max2) {
max2 = tmp;
}
if (tmp < 0) {
tmp = 0;
}
}
if (maxFinal < max1 + max2) {
maxFinal = max1 + max2;
}
}
return maxFinal;
}
Jerky Lu wechat
欢迎加入微信公众号